'''
createDate: 2024-03-26 15:38:01
Author: liangc
LastEditTime: 2024-03-26 15:56:50
LastEditors: zclee
FilePath: /quant/data_generator.py
Description: 生成假数据
'''

import numpy as np
for i in range(512):
    save_path = f'working/data/{str(i).zfill(4)}'
    img = np.random.random(size=[1, 3, 224, 224])
    np.save(file=save_path, arr=img)
print(f'数据生成完毕,并保存至:working/data')